
package etcdserver

import (
	"encoding/json"
	"fmt"
	"path"
	"strconv"
	"time"

	"github.com/coreos/go-semver/semver"
	"go.etcd.io/etcd/pkg/v3/pbutil"
	"go.etcd.io/etcd/server/v3/etcdserver/api"
	"go.etcd.io/etcd/server/v3/etcdserver/api/membership"
	"go.etcd.io/etcd/server/v3/etcdserver/api/v2store"

	"go.uber.org/zap"
)

const v2Version = "v2"

// ApplierV2是处理V2 raft消息的接口
// ApplierV2 接口中定义了处理 v2 版本消息的方法， 即 applyV2Request（） 方法中调用 的
// Post（）、 Put（）、 Delete（）、 QGet（） 、 Sync（） 等方法。结构体 applierV2store 实现 ApplierV2 接口 ，
// EtcdServer.applyV2 字段就是 applierV2store 类型。 applierV2store 中最重要的就是 store 字段，它
// 指向 当前节点的 v2 存储。
type ApplierV2 interface {
	Delete(r *RequestV2) Response
	Post(r *RequestV2) Response
	Put(r *RequestV2, shouldApplyV3 membership.ShouldApplyV3) Response
	QGet(r *RequestV2) Response
	Sync(r *RequestV2) Response
}

func NewApplierV2(lg *zap.Logger, s v2store.Store, c *membership.RaftCluster) ApplierV2 {
	if lg == nil {
		lg = zap.NewNop()
	}
	return &applierV2store{lg: lg, store: s, cluster: c}
}

type applierV2store struct {
	lg      *zap.Logger
	store   v2store.Store
	cluster *membership.RaftCluster
}

func (a *applierV2store) Delete(r *RequestV2) Response {
	switch {
	case r.PrevIndex > 0 || r.PrevValue != "":
		return toResponse(a.store.CompareAndDelete(r.Path, r.PrevValue, r.PrevIndex))
	default:
		return toResponse(a.store.Delete(r.Path, r.Dir, r.Recursive))
	}
}

func (a *applierV2store) Post(r *RequestV2) Response {
	return toResponse(a.store.Create(r.Path, r.Dir, r.Val, true, r.TTLOptions()))
}

// applierV2store.Put（） 方法 ， 其中根据请求内容进行分类处理
func (a *applierV2store) Put(r *RequestV2, shouldApplyV3 membership.ShouldApplyV3) Response {
	// 根据请求内 容创 建 TTLOptionSet 实例，其 中会设置节点的超时时间、此次请求是否 为 刷新操作
	ttlOptions := r.TTLOptions()
	exists, existsSet := pbutil.GetBool(r.PrevExist)	// 修改节点之前是否存在
	switch {
	case existsSet:
		if exists {
			if r.PrevIndex == 0 && r.PrevValue == "" {
				// 未提供 PreIndex 和 PreValue 信息，则直接调用 Update（）方法更新节点值
				return toResponse(a.store.Update(r.Path, r.Val, ttlOptions))
			}
			// 提供 了 PreIndex 和 PreValue 信息 ， 则 调用 CompareAndSwap （ ） 方 法更新节点位
			// CompareAndSwap （）方法会比较当前节点的 PreIndex 和 PreValue 是否与 此次操作提供提供
			// 的位相同，然后决定是否修改
			return toResponse(a.store.CompareAndSwap(r.Path, r.PrevValue, r.PrevIndex, r.Val, ttlOptions))
		}
		return toResponse(a.store.Create(r.Path, r.Dir, r.Val, false, ttlOptions))
	case r.PrevIndex > 0 || r.PrevValue != "":
		return toResponse(a.store.CompareAndSwap(r.Path, r.PrevValue, r.PrevIndex, r.Val, ttlOptions))
	default:
		// 操作的 节点是“/0/members ” 下的节点集群成员信息节点 ，注意，不会修改 v2 存储 中对应 的节点
		if storeMemberAttributeRegexp.MatchString(r.Path) {
			// 从节点的路径信息 中 解析得到节点 id
			id := membership.MustParseMemberIDFromKey(a.lg, path.Dir(r.Path))
			var attr membership.Attributes	// 将节点值反序列化
			if err := json.Unmarshal([]byte(r.Val), &attr); err != nil {
				a.lg.Panic("failed to unmarshal", zap.String("value", r.Val), zap.Error(err))
			}
			if a.cluster != nil {	// 更新 RaftCluster 中对应节点的信息
				a.cluster.UpdateAttributes(id, attr, shouldApplyV3)
			}
			// 由于没有使用者，因此返回空响应。
			return Response{}
		}
		// TODO删除etcd 3.6中为避免v2和v3之间冲突而设置的v2版本
		if r.Path == membership.StoreClusterVersionKey() {	// 操作的节点是“ /0/version”
			if a.cluster != nil {	// 更新 RaftCluster 中的版本信息
				// 持久化到后端给定的v2store可能非常陈旧
				a.cluster.SetVersion(semver.Must(semver.NewVersion(r.Val)), api.UpdateCapability, shouldApplyV3)
			}
			return Response{}
		}
		// 如果不是上述两种节点，则直接调用 Set （）方法更新对应节点值
		return toResponse(a.store.Set(r.Path, r.Dir, r.Val, ttlOptions))
	}
}

func (a *applierV2store) QGet(r *RequestV2) Response {
	return toResponse(a.store.Get(r.Path, r.Recursive, r.Sorted))
}

func (a *applierV2store) Sync(r *RequestV2) Response {
	a.store.DeleteExpiredKeys(time.Unix(0, r.Time))
	return Response{}
}

// applyV2Request将r解释为对v2store的调用。X 
// 并返回从v2store解释的响应。事件
func (s *EtcdServer) applyV2Request(r *RequestV2, shouldApplyV3 membership.ShouldApplyV3) (resp Response) {
	stringer := panicAlternativeStringer{
		stringer:    r,
		alternative: func() string { return fmt.Sprintf("id:%d,method:%s,path:%s", r.ID, r.Method, r.Path) },
	}
	defer func(start time.Time) {
		success := resp.Err == nil
		applySec.WithLabelValues(v2Version, r.Method, strconv.FormatBool(success)).Observe(time.Since(start).Seconds())
		warnOfExpensiveRequest(s.Logger(), s.Cfg.WarningApplyDuration, start, stringer, nil, nil)
	}(time.Now())

	switch r.Method {
	case "POST":
		return s.applyV2.Post(r)
	case "PUT":
		return s.applyV2.Put(r, shouldApplyV3)
	case "DELETE":
		return s.applyV2.Delete(r)
	case "QGET":
		return s.applyV2.QGet(r)
	case "SYNC":
		return s.applyV2.Sync(r)
	default:
		// 永远不应该达到此值，但只是以防万一：
		return Response{Err: ErrUnknownMethod}
	}
}

func (r *RequestV2) TTLOptions() v2store.TTLOptionSet {
	refresh, _ := pbutil.GetBool(r.Refresh)
	ttlOptions := v2store.TTLOptionSet{Refresh: refresh}
	if r.Expiration != 0 {
		ttlOptions.ExpireTime = time.Unix(0, r.Expiration)
	}
	return ttlOptions
}

func toResponse(ev *v2store.Event, err error) Response {
	return Response{Event: ev, Err: err}
}
